home *** CD-ROM | disk | FTP | other *** search
/ IRIX 5.3 for Indy R4400 / IRIX 5.3 for Indy R4400 175MHz.img / dist / eoe2.idb / usr / lib / perl / chat2.pl.z / chat2.pl
Perl Script  |  1995-02-28  |  9KB  |  343 lines
  1. ## chat.pl: chat with a server
  2. ## V2.01.alpha.7 91/06/16
  3. ## Randal L. Schwartz
  4.  
  5. package chat;
  6.  
  7. $sockaddr = 'S n a4 x8';
  8. chop($thishost = `hostname`); $thisaddr = (gethostbyname($thishost))[4];
  9. $thisproc = pack($sockaddr, 2, 0, $thisaddr);
  10.  
  11. # *S = symbol for current I/O, gets assigned *chatsymbol....
  12. $next = "chatsymbol000000"; # next one
  13. $nextpat = "^chatsymbol"; # patterns that match next++, ++, ++, ++
  14.  
  15.  
  16. ## $handle = &chat'open_port("server.address",$port_number);
  17. ## opens a named or numbered TCP server
  18.  
  19. sub open_port { ## public
  20.     local($server, $port) = @_;
  21.  
  22.     local($serveraddr,$serverproc);
  23.  
  24.     *S = ++$next;
  25.     if ($server =~ /^(\d+)+\.(\d+)\.(\d+)\.(\d+)$/) {
  26.         $serveraddr = pack('C4', $1, $2, $3, $4);
  27.     } else {
  28.         local(@x) = gethostbyname($server);
  29.         return undef unless @x;
  30.         $serveraddr = $x[4];
  31.     }
  32.     $serverproc = pack($sockaddr, 2, $port, $serveraddr);
  33.     # should really be something like:
  34.     # require 'sys/socket.ph';
  35.     # socket(S, &AF_INET, &SOCK_STREAM, ((getprotobyname('tcp'))[2]));
  36.     unless (socket(S, 2, 2, 6)) {
  37.         # XXX hardwired $AF_SOCKET, $SOCK_STREAM, 'tcp'
  38.         # but who the heck would change these anyway? (:-)
  39.         ($!) = ($!, close(S)); # close S while saving $!
  40.         return undef;
  41.     }
  42.     unless (bind(S, $thisproc)) {
  43.         ($!) = ($!, close(S)); # close S while saving $!
  44.         return undef;
  45.     }
  46.     unless (connect(S, $serverproc)) {
  47.         ($!) = ($!, close(S)); # close S while saving $!
  48.         return undef;
  49.     }
  50.     select((select(S), $| = 1)[0]);
  51.     $next; # return symbol for switcharound
  52. }
  53.  
  54. ## ($host, $port, $handle) = &chat'open_listen([$port_number]);
  55. ## opens a TCP port on the current machine, ready to be listened to
  56. ## if $port_number is absent or zero, pick a default port number
  57. ## process must be uid 0 to listen to a low port number
  58.  
  59. sub open_listen { ## public
  60.  
  61.     *S = ++$next;
  62.     local($thisport) = shift || 0;
  63.     local($thisproc_local) = pack($sockaddr, 2, $thisport, $thisaddr);
  64.     local(*NS) = "__" . time;
  65.     unless (socket(NS, 2, 1, 6)) {
  66.         # XXX hardwired $AF_SOCKET, $SOCK_STREAM, 'tcp'
  67.         # but who the heck would change these anyway? (:-)
  68.         ($!) = ($!, close(NS));
  69.         return undef;
  70.     }
  71.     unless (bind(NS, $thisproc_local)) {
  72.         ($!) = ($!, close(NS));
  73.         return undef;
  74.     }
  75.     unless (listen(NS, 1)) {
  76.         ($!) = ($!, close(NS));
  77.         return undef;
  78.     }
  79.     select((select(NS), $| = 1)[0]);
  80.     local($family, $port, @myaddr) =
  81.         unpack("S n C C C C x8", getsockname(NS));
  82.     $S{"needs_accept"} = *NS; # so expect will open it
  83.     (@myaddr, $port, $next); # returning this
  84. }
  85.  
  86. ## $handle = &chat'open_proc("command","arg1","arg2",...);
  87. ## opens a /bin/sh on a pseudo-tty
  88.  
  89. sub open_proc { ## public
  90.     local(@cmd) = @_;
  91.  
  92.     *S = ++$next;
  93.     local(*TTY) = "__TTY" . time;
  94.     local($pty,$tty) = &_getpty(S,TTY);
  95.     die "Cannot find a new pty" unless defined $pty;
  96.     local($pid) = fork;
  97.     die "Cannot fork: $!" unless defined $pid;
  98.     unless ($pid) {
  99.         close STDIN; close STDOUT; close STDERR;
  100.         setpgrp(0,$$);
  101.         if (open(DEVTTY, "/dev/tty")) {
  102.             ioctl(DEVTTY,0x20007471,0);        # XXX s/b &TIOCNOTTY
  103.             close DEVTTY;
  104.         }
  105.         open(STDIN,"<&TTY");
  106.         open(STDOUT,">&TTY");
  107.         open(STDERR,">&STDOUT");
  108.         die "Oops" unless fileno(STDERR) == 2;    # sanity
  109.         close(S);
  110.         exec @cmd;
  111.         die "Cannot exec @cmd: $!";
  112.     }
  113.     close(TTY);
  114.     $PID{$next} = $pid;
  115.     $next; # return symbol for switcharound
  116. }
  117.  
  118. # $S is the read-ahead buffer
  119.  
  120. ## $return = &chat'expect([$handle,] $timeout_time,
  121. ##     $pat1, $body1, $pat2, $body2, ... )
  122. ## $handle is from previous &chat'open_*().
  123. ## $timeout_time is the time (either relative to the current time, or
  124. ## absolute, ala time(2)) at which a timeout event occurs.
  125. ## $pat1, $pat2, and so on are regexs which are matched against the input
  126. ## stream.  If a match is found, the entire matched string is consumed,
  127. ## and the corresponding body eval string is evaled.
  128. ##
  129. ## Each pat is a regular-expression (probably enclosed in single-quotes
  130. ## in the invocation).  ^ and $ will work, respecting the current value of $*.
  131. ## If pat is 'TIMEOUT', the body is executed if the timeout is exceeded.
  132. ## If pat is 'EOF', the body is executed if the process exits before
  133. ## the other patterns are seen.
  134. ##
  135. ## Pats are scanned in the order given, so later pats can contain
  136. ## general defaults that won't be examined unless the earlier pats
  137. ## have failed.
  138. ##
  139. ## The result of eval'ing body is returned as the result of
  140. ## the invocation.  Recursive invocations are not thought
  141. ## through, and may work only accidentally. :-)
  142. ##
  143. ## undef is returned if either a timeout or an eof occurs and no
  144. ## corresponding body has been defined.
  145. ## I/O errors of any sort are treated as eof.
  146.  
  147. $nextsubname = "expectloop000000"; # used for subroutines
  148.  
  149. sub expect { ## public
  150.     if ($_[0] =~ /$nextpat/) {
  151.         *S = shift;
  152.     }
  153.     local($endtime) = shift;
  154.  
  155.     local($timeout,$eof) = (1,1);
  156.     local($caller) = caller;
  157.     local($rmask, $nfound, $timeleft, $thisbuf);
  158.     local($cases, $pattern, $action, $subname);
  159.     $endtime += time if $endtime < 600_000_000;
  160.  
  161.     if (defined $S{"needs_accept"}) { # is it a listen socket?
  162.         local(*NS) = $S{"needs_accept"};
  163.         delete $S{"needs_accept"};
  164.         $S{"needs_close"} = *NS;
  165.         unless(accept(S,NS)) {
  166.             ($!) = ($!, close(S), close(NS));
  167.             return undef;
  168.         }
  169.         select((select(S), $| = 1)[0]);
  170.     }
  171.  
  172.     # now see whether we need to create a new sub:
  173.  
  174.     unless ($subname = $expect_subname{$caller,@_}) {
  175.         # nope.  make a new one:
  176.         $expect_subname{$caller,@_} = $subname = $nextsubname++;
  177.  
  178.         $cases .= <<"EDQ"; # header is funny to make everything elsif's
  179. sub $subname {
  180.     LOOP: {
  181.         if (0) { ; }
  182. EDQ
  183.         while (@_) {
  184.             ($pattern,$action) = splice(@_,0,2);
  185.             if ($pattern =~ /^eof$/i) {
  186.                 $cases .= <<"EDQ";
  187.         elsif (\$eof) {
  188.              package $caller;
  189.             $action;
  190.         }
  191. EDQ
  192.                 $eof = 0;
  193.             } elsif ($pattern =~ /^timeout$/i) {
  194.             $cases .= <<"EDQ";
  195.         elsif (\$timeout) {
  196.              package $caller;
  197.             $action;
  198.         }
  199. EDQ
  200.                 $timeout = 0;
  201.             } else {
  202.                 $pattern =~ s#/#\\/#g;
  203.             $cases .= <<"EDQ";
  204.         elsif (\$S =~ /$pattern/) {
  205.             \$S = \$';
  206.              package $caller;
  207.             $action;
  208.         }
  209. EDQ
  210.             }
  211.         }
  212.         $cases .= <<"EDQ" if $eof;
  213.         elsif (\$eof) {
  214.             undef;
  215.         }
  216. EDQ
  217.         $cases .= <<"EDQ" if $timeout;
  218.         elsif (\$timeout) {
  219.             undef;
  220.         }
  221. EDQ
  222.         $cases .= <<'ESQ';
  223.         else {
  224.             $rmask = "";
  225.             vec($rmask,fileno(S),1) = 1;
  226.             ($nfound, $rmask) =
  227.                  select($rmask, undef, undef, $endtime - time);
  228.             if ($nfound) {
  229.                 $nread = sysread(S, $thisbuf, 1024);
  230.                 if ($nread > 0) {
  231.                     $S .= $thisbuf;
  232.                 } else {
  233.                     $eof++, redo LOOP; # any error is also eof
  234.                 }
  235.             } else {
  236.                 $timeout++, redo LOOP; # timeout
  237.             }
  238.             redo LOOP;
  239.         }
  240.     }
  241. }
  242. ESQ
  243.         eval $cases; die "$cases:\n$@" if $@;
  244.     }
  245.     $eof = $timeout = 0;
  246.     do $subname();
  247. }
  248.  
  249. ## &chat'print([$handle,] @data)
  250. ## $handle is from previous &chat'open().
  251. ## like print $handle @data
  252.  
  253. sub print { ## public
  254.     if ($_[0] =~ /$nextpat/) {
  255.         *S = shift;
  256.     }
  257.     print S @_;
  258. }
  259.  
  260. ## &chat'close([$handle,])
  261. ## $handle is from previous &chat'open().
  262. ## like close $handle
  263.  
  264. sub close { ## public
  265.     local($pid);
  266.     if ($_[0] =~ /$nextpat/) {
  267.         $pid = $PID{$_[0]};
  268.          *S = shift;
  269.     } else {
  270.         $pid = $PID{$next};
  271.     }
  272.     close(S);
  273.     waitpid($pid,0);
  274.     if (defined $S{"needs_close"}) { # is it a listen socket?
  275.         local(*NS) = $S{"needs_close"};
  276.         delete $S{"needs_close"};
  277.         close(NS);
  278.     }
  279. }
  280.  
  281. ## @ready_handles = &chat'select($timeout, @handles)
  282. ## select()'s the handles with a timeout value of $timeout seconds.
  283. ## Returns an array of handles that are ready for I/O.
  284. ## Both user handles and chat handles are supported (but beware of
  285. ## stdio's buffering for user handles).
  286.  
  287. sub select { ## public
  288.     local($timeout) = shift;
  289.     local(@handles) = @_;
  290.     local(%handlename) = ();
  291.     local(%ready) = ();
  292.     local($caller) = caller;
  293.     local($rmask) = "";
  294.     for (@handles) {
  295.         if (/$nextpat/o) { # one of ours... see if ready
  296.             local(*SYM) = $_;
  297.             if (length($SYM)) {
  298.                 $timeout = 0; # we have a winner
  299.                 $ready{$_}++;
  300.             }
  301.             $handlename{fileno($_)} = $_;
  302.         } else {
  303.             $handlename{fileno(/'/ ? $_ : "$caller\'$_")} = $_;
  304.         }
  305.     }
  306.     for (sort keys %handlename) {
  307.         vec($rmask, $_, 1) = 1;
  308.     }
  309.     select($rmask, undef, undef, $timeout);
  310.     for (sort keys %handlename) {
  311.         $ready{$handlename{$_}}++ if vec($rmask,$_,1);
  312.     }
  313.     sort keys %ready;
  314. }
  315.  
  316. # ($pty,$tty) = $chat'_getpty(PTY,TTY):
  317. # internal procedure to get the next available pty.
  318. # opens pty on handle PTY, and matching tty on handle TTY.
  319. # returns undef if can't find a pty.
  320.  
  321. sub _getpty { ## private
  322.     local($_PTY,$_TTY) = @_;
  323.     $_PTY =~ s/^([^']+)$/(caller)[$[]."'".$1/e;
  324.     $_TTY =~ s/^([^']+)$/(caller)[$[]."'".$1/e;
  325.     local($pty,$tty);
  326.     for $bank (112..127) {
  327.         next unless -e sprintf("/dev/pty%c0", $bank);
  328.         for $unit (48..57) {
  329.             $pty = sprintf("/dev/pty%c%c", $bank, $unit);
  330.             open($_PTY,"+>$pty") || next;
  331.             select((select($_PTY), $| = 1)[0]);
  332.             ($tty = $pty) =~ s/pty/tty/;
  333.             open($_TTY,"+>$tty") || next;
  334.             select((select($_TTY), $| = 1)[0]);
  335.             system "stty nl>$tty";
  336.             return ($pty,$tty);
  337.         }
  338.     }
  339.     undef;
  340. }
  341.  
  342. 1;
  343.